package com.dkd.manage.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.dkd.common.constant.DkdContants;
import com.dkd.common.exception.ServiceException;
import com.dkd.common.utils.DateUtils;
import com.dkd.manage.domain.Emp;
import com.dkd.manage.domain.Task;
import com.dkd.manage.domain.TaskDetails;
import com.dkd.manage.domain.dto.TaskDetailsDto;
import com.dkd.manage.domain.dto.TaskDto;
import com.dkd.manage.domain.vo.TaskVo;
import com.dkd.manage.domain.vo.VendingMachineVo;
import com.dkd.manage.mapper.*;
import com.dkd.manage.service.ITaskService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.time.Duration;
import java.util.*;
import java.util.stream.Collectors;

/**
 * 工单Service业务层处理
 *
 * @author xiaoyu
 * @date 2025-08-14
 */
@Slf4j
@Service
public class TaskServiceImpl implements ITaskService {

    @Autowired
    private TaskMapper taskMapper;

    @Autowired
    private TaskTypeMapper taskTypeMapper;

    @Autowired
    private VendingMachineMapper vendingMachineMapper;

    @Autowired
    private EmpMapper empMapper;

    @Autowired
    private RedisTemplate redisTemplate;

    @Autowired
    private TaskDetailsMapper taskDetailsMapper;

    /**
     * 查询工单
     *
     * @param taskId 工单主键
     * @return 工单
     */
    @Override
    public Task selectTaskByTaskId(Long taskId) {
        return taskMapper.selectTaskByTaskId(taskId);
    }

    /**
     * 查询工单列表
     *
     * @param task 工单
     * @return 工单
     */
    @Override
    public List<Task> selectTaskList(Task task) {
        return taskMapper.selectTaskList(task);
    }

    /**
     * 新增工单
     *
     * @param task 工单
     * @return 结果
     */
    @Override
    public int insertTask(Task task) {
        task.setCreateTime(DateUtils.getNowDate());
        return taskMapper.insertTask(task);
    }

    /**
     * 修改工单
     *
     * @param task 工单
     * @return 结果
     */
    @Override
    public int updateTask(Task task) {
        task.setUpdateTime(DateUtils.getNowDate());
        return taskMapper.updateTask(task);
    }

    /**
     * 批量删除工单
     *
     * @param taskIds 需要删除的工单主键
     * @return 结果
     */
    @Override
    public int deleteTaskByTaskIds(Long[] taskIds) {
        return taskMapper.deleteTaskByTaskIds(taskIds);
    }

    /**
     * 删除工单信息
     *
     * @param taskId 工单主键
     * @return 结果
     */
    @Override
    public int deleteTaskByTaskId(Long taskId) {
        return taskMapper.deleteTaskByTaskId(taskId);
    }

    /**
     * 查询工单列表
     *
     * @param task 工单
     * @return 工单集合
     */
    @Override
    public List<TaskVo> selectTaskVoList(Task task) {
        List<Integer> productTypeIds = new ArrayList<>();
        String flag = (String) task.getParams().get("isRepair");
        // 工单状态 1 表示投放工单 2 表示补货工单 3 表示维修工单 4 撤机工单
        // 如果传递的是true则表示查询 工单类型id为 1,3,4的工单
        if ("true".equals(flag)) {
            Collections.addAll(productTypeIds, DkdContants.DEPLOY_TASK, DkdContants.REPAIR_TASK, DkdContants.REVOKE_TASK);
        } else if ("false".equals(flag)) {
            // 如果传递的是false则表示查询 工单类型id为 2的工单
            Collections.addAll(productTypeIds, DkdContants.SUPPLY_TASK);
        }
        List<TaskVo> taskVoList = taskMapper.selectTaskVoList(task, productTypeIds);
        taskVoList.forEach(taskVo -> taskVo.setTaskType(taskTypeMapper.selectTaskTypeByTypeId(taskVo.getProductTypeId())));

        return taskVoList;
    }

    /**
     * 新增运营运维工单
     *
     * @param taskDto 工单对象
     * @return 结果
     */
    @Transactional(rollbackFor = Exception.class)
    @Override
    public int insertTaskDto(TaskDto taskDto) {
        // 1.查询售货机是否存在
        VendingMachineVo vendingMachineVo = vendingMachineMapper.selectVendingMachineByInnerCode(taskDto.getInnerCode());
        if (vendingMachineVo == null) {
            throw new ServiceException("售货机不存在");
        }
        // 2.校验售货机状态与工单类型是否相符合
        checkCreateType(vendingMachineVo.getVmStatus(), taskDto.getProductTypeId());
        // 3.检查该设备是否有未完成的同类型工单
        hasTask(taskDto);
        // 4.查询并校验员工是否存在
        Emp emp = empMapper.selectEmpById(taskDto.getUserId());
        if (emp == null) {
            throw new ServiceException("员工不存在");
        }
        // 5.校验员工区域是否匹配
        if (!emp.getRegionId().equals(vendingMachineVo.getRegionId())) {
            throw new ServiceException("员工区域不匹配,无法处理此工单");
        }
        // 6.将dto转为po,并且补充属性保存工单
        Task task = BeanUtil.copyProperties(taskDto, Task.class);
        task.setTaskStatus(DkdContants.TASK_STATUS_CREATE); // 设置工单状态
        task.setUserName(emp.getUserName()); // 设置执行人名称
        task.setRegionId(vendingMachineVo.getRegionId()); // 设置所属区域id
        task.setAddr(vendingMachineVo.getAddr()); // 设置售货机地址
        task.setCreateTime(DateUtils.getNowDate()); // 设置创建时间
        task.setTaskCode(generateTaskCode()); // 设置工单编号
        int result = taskMapper.insertTask(task);

        // 7.判断是否是补货工单
        if (taskDto.getProductTypeId().equals(DkdContants.TASK_TYPE_SUPPLY)) {
            // 获取补货工单详情
            List<TaskDetailsDto> details = taskDto.getDetails();
            if (CollUtil.isEmpty(details)) {
                throw new ServiceException("补货工单详情不能为空");
            }
            // 将dto转为po
            List<TaskDetails> taskDetailsList = details.stream().map(dto -> {
                TaskDetails taskDetails = BeanUtil.copyProperties(dto, TaskDetails.class);
                taskDetails.setTaskId(task.getTaskId());
                return taskDetails;
            }).collect(Collectors.toList());

            // 批量插入工单详情
            taskDetailsMapper.insertBatchTaskDetails(taskDetailsList);
        }
        return result;
    }

    /**
     * 取消工单
     *
     * @param task 工单对象
     * @return 结果
     */
    @Override
    public int cancelTask(Task task) {
        // 1.判断工单状态是否可以取消
        Task taskDb = taskMapper.selectTaskByTaskId(task.getTaskId());
        // 判断工单状态是否为已取消，如果是则抛出异常
        if (taskDb.getTaskStatus().equals(DkdContants.TASK_STATUS_CANCEL)) {
            throw new ServiceException("工单已取消，不能再次取消");
        }
        // 判断工单状态是否为已完成，如果是则抛出异常
        if (taskDb.getTaskStatus().equals(DkdContants.TASK_STATUS_FINISH)) {
            throw new ServiceException("工单已完成，不能取消");
        }

        // 2.设置更新字段
        task.setTaskStatus(DkdContants.TASK_STATUS_CANCEL); // 设置工单状态为已取消
        task.setUpdateTime(DateUtils.getNowDate());

        // 3.更新工单
        return taskMapper.updateTask(task);
    }

    /**
     * 生成工单编号
     *
     * @return 工单编号
     */
    public String generateTaskCode() {
        // 获取当前日期并格式化为yyyyMMdd
        String dateStr = DateUtils.getDate().replaceAll("-", "");
        // 根据日期生成redis的键
        String key = "dkd.task.code" + dateStr;
        // 判断这个key在redis中是否存在
        if (!redisTemplate.hasKey(key)) {
            // 如果key不存在设置初始值为1，并且指定过期时间为1天
            redisTemplate.opsForValue().set(key, 1, Duration.ofDays(1));
            return dateStr + "0001";
        }
        // 如果key存在,计数器+1，但是要确保字符串的长度要保持为4位
        // padPre第一个参数表示要填充的字符串，第二个参数表示填充的长度，第三个参数表示如果长度不够填充的字符
        return dateStr + StrUtil.padPre(redisTemplate.opsForValue().increment(key).toString(), 4, "0");
    }

    /**
     * 检查工单是否存在
     *
     * @param taskDto 工单dto
     */
    private void hasTask(TaskDto taskDto) {
        Task taskParam = new Task();
        taskParam.setInnerCode(taskDto.getInnerCode());
        taskParam.setProductTypeId(taskDto.getProductTypeId());
        taskParam.setTaskStatus(DkdContants.TASK_STATUS_PROGRESS);
        List<Task> taskList = taskMapper.selectTaskList(taskParam);
        if (taskList != null && !taskList.isEmpty()) {
            throw new ServiceException("该设备有未完成的工单,请勿重复创建");
        }
    }

    /**
     * 校验工单状态与售货机的状态是否相符合
     *
     * @param vmStatus      售货机状态
     * @param productTypeId 工单类型
     */
    public void checkCreateType(Long vmStatus, Long productTypeId) {
        // 如果是投放工单,说明设备在运行中,抛出异常
        if (Objects.equals(productTypeId, DkdContants.TASK_TYPE_DEPLOY) && Objects.equals(vmStatus, DkdContants.VM_STATUS_RUNNING)) {
            throw new ServiceException("设备正在运行中,无法投放");
        }
        // 如果是维修工单,设备不在运行中,抛出异常
        if (Objects.equals(productTypeId, DkdContants.TASK_TYPE_REPAIR) && !Objects.equals(vmStatus, DkdContants.VM_STATUS_RUNNING)) {
            throw new ServiceException("设备不在运行中,无法维修");
        }
        // 如果是补货工单,设备不在运行中,抛出异常
        if (Objects.equals(productTypeId, DkdContants.TASK_TYPE_SUPPLY) && !Objects.equals(vmStatus, DkdContants.VM_STATUS_RUNNING)) {
            throw new ServiceException("设备不在运行中,无法补货");
        }
        // 如果是撤机工单,设备不在运行中,抛出异常
        if (Objects.equals(productTypeId, DkdContants.TASK_TYPE_REVOKE) && !Objects.equals(vmStatus, DkdContants.VM_STATUS_RUNNING)) {
            throw new ServiceException("设备不在运行中,无法撤机");
        }
    }


    @Override
    public Integer countCompletedTasksByEmp(Long empId) {
        try {
            return taskMapper.countCompletedTasksByEmp(empId);
        } catch (Exception e) {
            log.error("统计员工已完成任务数量异常: {}", e.getMessage());
            return 0;
        }
    }

    @Override
    public Integer countCurrentTasksByEmp(Long empId) {
        try {
            return taskMapper.countCurrentTasksByEmp(empId);
        } catch (Exception e) {
            log.error("统计员工当前任务数量异常: {}", e.getMessage());
            return 0;
        }
    }

    @Override
    public Integer countTotalTasksByEmp(Long empId) {
        try {
            return taskMapper.countTotalTasksByEmp(empId);
        } catch (Exception e) {
            log.error("统计员工总任务数量异常: {}", e.getMessage());
            return 0;
        }
    }

    @Override
    public Integer countSuccessfulTasksByEmp(Long empId) {
        try {
            return taskMapper.countSuccessfulTasksByEmp(empId);
        } catch (Exception e) {
            log.error("统计员工成功任务数量异常: {}", e.getMessage());
            return 0;
        }
    }

    @Override
    public Set<String> getSkilledMachinesByEmp(Long empId) {
        try {
            return taskMapper.getSkilledMachinesByEmp(empId);
        } catch (Exception e) {
            log.error("获取员工熟悉设备异常: {}", e.getMessage());
            return Set.of();
        }
    }

    @Override
    public Double getAvgCompletionTimeByEmp(Long empId) {
        try {
            Double avgTime = taskMapper.getAvgCompletionTimeByEmp(empId);
            return avgTime != null ? avgTime : 4.0;
        } catch (Exception e) {
            log.error("获取员工平均完成时间异常: {}", e.getMessage());
            return 4.0;
        }
    }

    @Override
    public Double getAvgQualityScoreByEmp(Long empId) {
        try {
            Double avgScore = taskMapper.getAvgQualityScoreByEmp(empId);
            return avgScore != null ? avgScore : 5.0;
        } catch (Exception e) {
            log.error("获取员工平均质量评分异常: {}", e.getMessage());
            return 5.0;
        }
    }

    @Override
    public Integer countTodayTasksByEmp(Long empId) {
        try {
            return taskMapper.countTodayTasksByEmp(empId);
        } catch (Exception e) {
            log.error("统计员工今日任务异常: {}", e.getMessage());
            return 0;
        }
    }

    @Override
    public Double getTaskCompletionRate(Long empId) {
        try {
            Integer total = countTotalTasksByEmp(empId);
            Integer completed = countCompletedTasksByEmp(empId);
            return total > 0 ? (double) completed / total : 0.8;
        } catch (Exception e) {
            log.error("计算任务完成率异常: {}", e.getMessage());
            return 0.8;
        }
    }
}
